api/distributor_views.py

@@ -3,6 +3,7 @@

                 from __future__ import division
              

                 
              

                 from django_logit import logit
              

                +from django_models_ext import ProvinceShortModelMixin
              

                 from django_response import response
              

                 from paginator import pagination
              

                 
              
@@ -17,6 +18,10 @@ def distributor_add(request):

                     province_code = request.POST.get('province_code', '')
              

                     province_name = request.POST.get('province_name', '')
              

                 
              

                +    # 如果省份编码不存在，而省份名称存在
              

                +    if not province_code and province_name:
              

                +        province_code = ProvinceShortModelMixin.PROVINCE_NAME_CODE_DICT.get(province_name, '')
              

                +
              

                     admin_id = request.session.get('admin_id')
              

                 
              

                     try:
              
@@ -85,6 +90,9 @@ def distributor_update(request):

                         distributor.distributor_province_code = province_code
              

                     if province_name:
              

                         distributor.distributor_province_name = province_name
              

                +        # 如果省份编码不存在，而省份名称存在
              

                +        if not province_code:
              

                +            distributor.distributor_province_code = ProvinceShortModelMixin.PROVINCE_NAME_CODE_DICT.get(province_name, '')
              

                 
              

                     distributor.save()